import qrcode
from PIL import Image

# 创建QRCode对象
qr = qrcode.QRCode(
    version=1,
    error_correction=qrcode.constants.ERROR_CORRECT_L,
    box_size=10,
    border=4,
)

# 添加数据到QRCode对象
data = "https://www.example.com"
qr.add_data(data)
qr.make(fit=True)

# 创建图像
img = qr.make_image(fill_color="black", back_color="white")

# 加载logo图片
logo = Image.open('2.png')  # 这里替换为你的logo文件名

# 缩放logo到合适的尺寸
max_size = 50  # logo的最大宽度和高度
if logo.width > max_size or logo.height > max_size:
    logo = logo.resize((max_size, max_size), Image.Resampling.LANCZOS)

print(img.width)
print(img.height)
print(logo.width)
print(logo.height)
# 计算logo的位置
pos = ((img.height - logo.width) // 2, (img.height - logo.height) // 2)

# 将logo粘贴到二维码中心
img.paste(logo, pos, mask=logo)

# 保存图像
img.save("example.png")
